# llm_engine/config.py
import os

# 获取llm_engine目录的绝对路径
LLM_ENGINE_ROOT_PATH = os.path.dirname(os.path.abspath(__file__))

# 定义ChromaDB的存储路径，它将位于llm_engine目录内部
CHROMA_DB_PATH = os.path.join(LLM_ENGINE_ROOT_PATH, "chroma_db_storage")


API_KEY = "YOUR KEY" # 替换成您自己的Key

# OpenAI兼容API端点
API_BASE = "https://open.bigmodel.cn/api/paas/v4/"

# 指定要使用的模型名称
# 用于生成最终答案的聊天/补全模型
LLM_MODEL_NAME = "glm-4-plus"  # 或者 moonshot-v1-32k, moonshot-v1-128k
# 用于将文本转换为向量的Embedding模型
EMBEDDING_MODEL_NAME = "embedding-2" # 已通过客服确认月之暗面没有embedding模型, 换智谱.
# EMBEDDING_MODEL_NAME = "moonshot-v1-8k"